package com.zee.wechat.framework.utils;

import org.springframework.beans.BeanUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.function.Supplier;

/**
 * 转换对象工具
 *
 * @author Zee
 * @date 2020年6月19日
 */
public class BeanConvertUtils extends BeanUtils {
    /**
     * 转换对象
     *
     * @param source         源对象
     * @param targetSupplier 目标对象供应方
     * @param <S>            源对象类型
     * @param <T>            目标对象类型
     * @return 目标对象
     */
    public static <S, T> T convertTo(S source, Supplier<T> targetSupplier) {
        return convertTo(source, targetSupplier, null);
    }

    /**
     * 转换对象带回调
     *
     * @param source         源对象
     * @param targetSupplier 目标对象供应方
     * @param callBack       回调方法
     * @param <S>            源对象类型
     * @param <T>            目标对象类型
     * @return 目标对象
     */
    public static <S, T> T convertTo(S source, Supplier<T> targetSupplier, ConvertCallBack<S, T> callBack) {
        if (Objects.isNull(source) || Objects.isNull(targetSupplier)) {
            return null;
        }
        T target = targetSupplier.get();
        copyProperties(source, target);
        if (Objects.nonNull(callBack)) {
            callBack.callBack(source, target);
        }
        return target;
    }

    /**
     * 转换list对象
     *
     * @param sources        源对象list
     * @param targetSupplier 目标对象供应方
     * @param <S>            源对象类型
     * @param <T>            目标对象类型
     * @return 目标对象list
     */
    public static <S, T> List<T> convertListTo(List<S> sources, Supplier<T> targetSupplier) {
        return convertListTo(sources, targetSupplier, null);
    }

    /**
     * 转换list对象带回调
     *
     * @param sources        源对象list
     * @param targetSupplier 目标对象供应方
     * @param callBack       回调方法
     * @param <S>            源对象类型
     * @param <T>            目标对象类型
     * @return 目标对象list
     */
    public static <S, T> List<T> convertListTo(List<S> sources, Supplier<T> targetSupplier, ConvertListCallBack<S, T> callBack) {
        if (Objects.isNull(sources) || Objects.isNull(targetSupplier)) {
            return null;
        }
        List<T> list = new ArrayList<>(sources.size());
        int index = 0;
        for (S source : sources) {
            T target = targetSupplier.get();
            copyProperties(source, target);
            if (Objects.nonNull(callBack)) {
                callBack.callBack(index, source, target);
            }
            list.add(target);
            index++;
        }
        return list;
    }

    @FunctionalInterface
    public interface ConvertCallBack<S, T> {
        /**
         * 回调方法
         *
         * @param s source
         * @param t target
         */
        void callBack(S s, T t);
    }

    @FunctionalInterface
    public interface ConvertListCallBack<S, T> {
        /**
         * 回调方法
         *
         * @param s     source
         * @param t     target
         * @param index index
         */
        void callBack(Integer index, S s, T t);
    }
}